package algorithm7.leetcode.每日一题.YEAR2023.month5;

import org.junit.jupiter.api.Test;

import java.util.ArrayList;

/**
 * @author merrick.peng
 *
 *          2455. 可被三整除的偶数的平均值
 */
public class DAY20230529_LC2455 {

    public int averageValue(int[] nums) {
//        return handle1(nums);
        return handle2(nums);
    }

    private int handle2(int[] nums) {
        // 次数
        int time = 0;
        int total = 0;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] % 6 == 0) {
                time++;
                total+=nums[i];
            }
        }
        return time == 0 ? 0 : total / time;
    }

    private int handle1(int[] nums) {
        int n = nums.length;
        ArrayList<Integer> list = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            int num = nums[i];
            if (num % 2 == 0 && num % 3 == 0) {
                list.add(num);
            }
        }
        return Math.floorDiv(list.stream().mapToInt(Integer::intValue).sum(), list.size() > 0 ? list.size() : 1);
    }

    /*

    给你一个由正整数组成的整数数组 nums ，返回其中可被 3 整除的所有偶数的平均值。
    注意：n 个元素的平均值等于 n 个元素 求和 再除以 n ，结果 向下取整 到最接近的整数。

    示例 1：
    输入：nums = [1,3,6,10,12,15]
    输出：9
    解释：6 和 12 是可以被 3 整除的偶数。(6 + 12) / 2 = 9 。

     */

    @Test
    void t() {
//        int[] nums = {1,3,6,10,12,15};
        int[] nums = {1,2,4,7,10};
        int res = averageValue(nums);
        System.out.println("res = " + res);
    }

}
